﻿#pragma once
#include"RBTree.h"
namespace bit
{
	template<class K, class V>
	class map
	{
		// 仿函数（函数对象）
		// 功能：从pair<K, V>中提取key值
		// operator()接收一个pair<K, V>类型的参数，返回其first成员（即key）
		struct MapKeyOfT
		{
			const K& operator()(const pair<K, V>& kv)
			{
				return kv.first;
			}
		};
	public:
		typedef typename RBTree<K, pair<const K, V>, MapKeyOfT>::iterator iterator;
		typedef typename RBTree<K, pair<const K, V>, MapKeyOfT>::const_iterator const_iterator;

		typedef typename RBTree<K, pair<const K, V>, MapKeyOfT>::reverse_iterator reverse_iterator;
		typedef typename RBTree<K, pair<const K, V>, MapKeyOfT>::const_reverse_iterator const_reverse_iterator;

		//复用
		iterator begin()
		{
			return _t.begin();
		}

		//同理
		iterator end()
		{
			return _t.end();
		}

		//同理
		const_iterator cbegin() const
		{
			return _t.cbegin();
		}

		//同理
		const_iterator cend() const
		{
			return _t.cend();
		}

		//同理
		reverse_iterator rbegin()
		{
			return _t.rbegin();
		}

		//同理
		reverse_iterator rend()
		{
			return _t.rend();
		}

		//同理
		const_reverse_iterator crbegin() const
		{
			return _t.crbegin();
		}

		//同理
		const_reverse_iterator crend() const
		{
			return _t.crend();
		}


		//同理
		pair<iterator, bool> insert(const pair<K, V>& kv)
		{
			return _t.Insert(kv);
		}

		//同理
		iterator find(const K& key)
		{
			return _t.Find(key);
		}

		//无法复用，因为这是map自己独有的东西，写在map类中
		//[]的意义： 
		//1：若待插入元素的键值key在map当中不存在，则insert函数插入成功，并返回一个pair对象，该对象包含插入后元素的迭代器和true。
		//2：若待插入元素的键值key在map当中已经存在，则insert函数插入失败，并返回一个pair对象，该对象包含map当中已经存在的键值为key元素的迭代器和false。
		V& operator[](const K& key)
		{
			pair<iterator, bool> ret = insert(make_pair(key, V()));
			return ret.first->second;
		}

	private:
		RBTree<K, pair<const K, V>, MapKeyOfT> _t;
	};

	//------------------------------------------------------------------
	//测试用例

	//测试map用iterator
	//效果：K无法修改，V可以
	void test_map1()
	{
		map<int, int> m;
		int a[] = { 4, 2, 6, 1, 3, 5, 15, 7, 16, 14 };
		for (auto e : a)
		{
			m.insert(make_pair(e, e));
		}

		map<int, int>::iterator it = m.begin();
		while (it != m.end())
		{
			//it->first += 100;//放开注释则报错
			it->second += 100;//但能修改V

			cout << it->first << ":" << it->second << endl;
			++it;
		}
		cout << endl;
	}

	//测试map用:const_iterator
	//效果：K无法修改，V也不可以
	void test_map2()
	{
		map<int, int> m;
		int a[] = { 4, 2, 6, 1, 3, 5, 15, 7, 16, 14 };
		for (auto e : a)
		{
			m.insert(make_pair(e, e));
		}

		map<int, int>::const_iterator it = m.cbegin();
		while (it != m.cend())
		{
			//it->first += 100;//放开注释则报错
			//it->second += 100;//放开注释则报错

			cout << it->first << ":" << it->second << endl;
			++it;
		}
		cout << endl;
	}

	//测试map用:reverse_iterator
	//效果：倒着遍历，K无法修改，V可以
	void test_map3()
	{
		map<int, int> m;
		int a[] = { 4, 2, 6, 1, 3, 5, 15, 7, 16, 14 };
		for (auto e : a)
		{
			m.insert(make_pair(e, e));
		}

		map<int, int>::reverse_iterator it = m.rbegin();
		while (it != m.rend())
		{
			//it->first += 100;//放开注释则报错
			it->second += 100;//但能修改V

			cout << it->first << ":" << it->second << endl;
			++it;
		}
		cout << endl;
	}

	//测试map用:const_reverse_iterator
	//效果：倒着遍历，K无法修改，V也不可以
	void test_map4()
	{
		map<int, int> m;
		int a[] = { 4, 2, 6, 1, 3, 5, 15, 7, 16, 14 };
		for (auto e : a)
		{
			m.insert(make_pair(e, e));
		}

		map<int, int>::const_reverse_iterator it = m.crbegin();
		while (it != m.crend())
		{
			//it->first += 100;//放开注释则报错
			//it->second += 100;//放开注释则报错

			cout << it->first << ":" << it->second << endl;
			++it;
		}
		cout << endl;
	}

	//测试map的[]
	void test_map5()
	{
		string arr[] = { "苹果", "西瓜", "苹果", "西瓜", "苹果", "苹果", "西瓜",
					 "苹果", "香蕉", "苹果", "西瓜", "香蕉", "草莓" };
		map<string, int> countMap;
		for (auto& e : arr)
		{

			countMap[e]++;
		}

		for (auto& kv : countMap)
		{
			cout << kv.first << ":" << kv.second << endl;
		}
		cout << endl;
	}
};